
from langdev_helper.llm.qwen import llm, llm0
from langdev_helper.prompt.word import get_word_prompt, get_word_list
from langdev_core.llms.qianfan import call_llm
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder

system_prompt = get_word_prompt(4)
user_prompt = "/cnt"
user_prompt = "/all"
# user_prompt = "/last"

# user_prompt+= get_word_list(3)

print(system_prompt)
print(user_prompt)

messages = [
    SystemMessage(content=system_prompt),
    HumanMessage(content=user_prompt)
]

# res = llm.invoke(messages)
# print(res.content)

call_llm(user_prompt, system_prompt)

